package com.fas.core.plugin.jfinal.generator;

import javax.sql.DataSource;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.Prop;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.c3p0.C3p0Plugin;
/**
 * 
 * @ClassName: CodeGenerator 
 * @Description: 代码生成器
 * 		<p> 主要生成Jfinal baseModel和Model</p>
 * @Copyright: Copyright (c) 2016 FAS 
 * @anthor yixiaoping<thinker1013@163.com>
 * @since 2016年1月24日 下午11:00:59
 * @version V1.0
 */
public class CodeGenerator {

	public static DataSource getDataSource() {
		Prop p = PropKit.use("appConfig.properties");
		C3p0Plugin c3p0Plugin = new C3p0Plugin(p.get("jdbc.url"),
				p.get("jdbc.username"), p.get("jdbc.password"));
		c3p0Plugin.start();
		return c3p0Plugin.getDataSource();
	}

	public static void main(String[] args) {
		generateSystemModule();
		generateConfigModule();
	}
	private static void generateConfigModule() {
		// base model 所使用的包名
		String baseModelPackageName = " com.fas.hehe.config.model.base";
		// base model 文件保存路径
		String baseModelOutputDir = PathKit.getWebRootPath()
				+ "/generator/src/com/fas/hehe/config/model/base";

		// model 所使用的包名 (MappingKit 默认使用的包名)
		String modelPackageName = " com.fas.hehe.config";
		// model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
		String modelOutputDir = baseModelOutputDir + "/..";

		// 创建生成器
		Generator gernerator = new Generator(getDataSource(),
				baseModelPackageName, baseModelOutputDir, modelPackageName,
				modelOutputDir);
		// 设置数据库方言
		gernerator.setDialect(new MysqlDialect());
		// 添加不需要生成的表名
		// gernerator.addExcludedTable("adv");
		// 设置是否在 Model 中生成 dao 对象
		gernerator.setGenerateDaoInModel(true);
		// 设置是否生成字典文件
		gernerator.setGenerateDataDictionary(true);
		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为
		// "User"而非 OscUser
		gernerator.setRemovedTableNamePrefixes("config_");
		gernerator.addExcludedTable("");
		// 生成
		gernerator.generate();
	}
	private static void generateSystemModule() {
		// base model 所使用的包名
		String baseModelPackageName = " com.fas.hehe.system.model.base";
		// base model 文件保存路径
		String baseModelOutputDir = PathKit.getWebRootPath()
				+ "/generator/src/com/fas/hehe/system/model/base";

		// model 所使用的包名 (MappingKit 默认使用的包名)
		String modelPackageName = " com.fas.hehe.system";
		// model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
		String modelOutputDir = baseModelOutputDir + "/..";

		// 创建生成器
		Generator gernerator = new Generator(getDataSource(),
				baseModelPackageName, baseModelOutputDir, modelPackageName,
				modelOutputDir);
		// 设置数据库方言
		gernerator.setDialect(new MysqlDialect());
		// 添加不需要生成的表名
		// gernerator.addExcludedTable("adv");
		// 设置是否在 Model 中生成 dao 对象
		gernerator.setGenerateDaoInModel(true);
		// 设置是否生成字典文件
		gernerator.setGenerateDataDictionary(true);
		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为
		// "User"而非 OscUser
		gernerator.setRemovedTableNamePrefixes("sys_");
		// 生成
		gernerator.generate();
	}
	
}
